

from functools import wraps
import random
import string

from flask import g
from flask import make_response, request

from ..config import BaseConfig
import os, base64
import re
from werkzeug.utils import secure_filename
import datetime
from flask import current_app
import time
import hashlib
from bs4 import BeautifulSoup


def allow_cross_domain(fun):
    @wraps(fun)
    def wrapper_fun(*args, **kwargs):
        allow_headers = "Referer,Accept,Origin,User-Agent,Content-Type,Authorization"
        resp = None
        if request.method == "OPTIONS":
            resp = make_response()
        else:
            result_json = fun(*args, **kwargs)
            resp = make_response(result_json)
        resp.headers['Access-Control-Allow-Origin'] = '*'
        resp.headers['Access-Control-Allow-Methods'] = 'PUT,GET,POST,DELETE'
        resp.headers['Access-Control-Allow-Headers'] = allow_headers
        # print("wrapper_fun",request.method,request.json,resp.headers,resp)
        return resp

    # print("wrapper_fun end")
    return wrapper_fun


def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1] in BaseConfig.ALLOWED_EXTENSIONS


# 判断手机号
def phonecheck(s):
    # 号码前缀，如果运营商启用新的号段，只需要在此列表将新的号段加上即可。
    phoneprefix = ['130', '131', '132', '133', '134', '135', '136', '137', '138', '139', '150', '151', '152', '153',
                   '155', '156', '158', '159', '170', '180', '183', '182', '185', '186', '187', '188', '189', '177']
    # 检测号码是否长度是否合法。
    if len(s) != 11:
        return False
    else:
        # 检测输入的号码是否全部是数字。
        if s.isdigit():
            # 检测前缀是否是正确。
            if s[:3] in phoneprefix:
                return True
            else:
                return False
        else:
            return False


# 把html_string的字符串转成文件保存，同时把对应的img src 改成链接, 后续增加其它操作
def translate(html_string):
    if html_string is None or len(html_string) == 0:
        print('translate html_string is error')
        return
    _dom = BeautifulSoup("<html>"+html_string+"</html>")
    for ele in _dom.find_all('img'):
        # base64 字符串变成图片保存
        _src_string = ele['src']
        print("translate _src_string", ele,ele['src'])
        if _src_string is not None and _src_string[0:4] != 'http':
            # 是src-data
            _position = _src_string.index(',')
            _front_string = _src_string[0:_position]
            _base64_string = _src_string[_position + 1:]
            _suffix_string_front_pos = _front_string.index('/')
            _suffix_string_back_pos = _front_string.index(';')
            _suffix_string = _front_string[_suffix_string_front_pos + 1:_suffix_string_back_pos]
            imgdata = base64.standard_b64decode(_base64_string)
            filename = str(int(time.time() * 1000)) + '.' + _suffix_string
            relative_file_name = os.path.join(current_app.config.get('UPLOAD_FOLDER'), filename)
            file_path = os.path.join(current_app.root_path, relative_file_name)
            file = open(file_path, 'wb')
            file.write(imgdata)
            file.close()
            # print('file_path',file_path)
            ele['src'] = os.path.join(current_app.config['HOST_URL'], relative_file_name)
            _style_string = ele.style if ele.style != None and len(ele.style) > 0 else ''
            _style_string = _style_string + 'width:100%;'
            ele['style']=_style_string
        else:
            _style_string = ele.style if ele.style != None and len(ele.style) > 0 else ''
            _style_string = _style_string + 'width:100%;'
            ele['style'] = _style_string
    print("translate basdf",str(_dom))
    return str(_dom)
    pass

# 往image下存文件
def save_file_to_image(file, file_path_name):
    if file and allowed_file(file.filename):
        relative_file_name = os.path.join("static/image/", file_path_name)
        absolute_file_path = os.path.join(current_app.root_path, relative_file_name)
        if os.path.isfile(absolute_file_path):
            os.remove(absolute_file_path)
        file.save(absolute_file_path)
        # 返回绝对url
        # print("save_file_to_image",current_app.config.get('HOST_URL'))
        # return current_app.config.get('HOST_URL') + relative_file_name
        return relative_file_name,absolute_file_path #保存相对路径更好
    return None
    pass


# 把一行数据库查询结果解析成dict
def row_to_dict(data):
    _dict = data.__dict__
    if _dict.get("_sa_instance_state"):
        _dict.pop('_sa_instance_state', None)
    return _dict

# 把多行数据库查询结果解析成python list 对象
def rows_to_list(data):
    list_zone = list()
    for value in data:
        list_zone.append(row_to_dict(value))
    return list_zone

def rows_to_dict(data):
    dict_zone = dict()
    for value in data:
        dict_zone[value.id]=row_to_dict(value)
    return dict_zone

#判断网站来自mobile还是pc
def checkMobile(request):
    """
    demo :
        @app.route('/m')
        def is_from_mobile():
            if checkMobile(request):
                return 'mobile'
            else:
                return 'pc'
    :param request:
    :return:
    """
    userAgent = request.headers['User-Agent']
    # userAgent = env.get('HTTP_USER_AGENT')

    _long_matches = r'googlebot-mobile|android|avantgo|blackberry|blazer|elaine|hiptop|ip(hone|od)|kindle|midp|mmp|mobile|o2|opera mini|palm( os)?|pda|plucker|pocket|psp|smartphone|symbian|treo|up\.(browser|link)|vodafone|wap|windows ce; (iemobile|ppc)|xiino|maemo|fennec'
    _long_matches = re.compile(_long_matches, re.IGNORECASE)
    _short_matches = r'1207|6310|6590|3gso|4thp|50[1-6]i|770s|802s|a wa|abac|ac(er|oo|s\-)|ai(ko|rn)|al(av|ca|co)|amoi|an(ex|ny|yw)|aptu|ar(ch|go)|as(te|us)|attw|au(di|\-m|r |s )|avan|be(ck|ll|nq)|bi(lb|rd)|bl(ac|az)|br(e|v)w|bumb|bw\-(n|u)|c55\/|capi|ccwa|cdm\-|cell|chtm|cldc|cmd\-|co(mp|nd)|craw|da(it|ll|ng)|dbte|dc\-s|devi|dica|dmob|do(c|p)o|ds(12|\-d)|el(49|ai)|em(l2|ul)|er(ic|k0)|esl8|ez([4-7]0|os|wa|ze)|fetc|fly(\-|_)|g1 u|g560|gene|gf\-5|g\-mo|go(\.w|od)|gr(ad|un)|haie|hcit|hd\-(m|p|t)|hei\-|hi(pt|ta)|hp( i|ip)|hs\-c|ht(c(\-| |_|a|g|p|s|t)|tp)|hu(aw|tc)|i\-(20|go|ma)|i230|iac( |\-|\/)|ibro|idea|ig01|ikom|im1k|inno|ipaq|iris|ja(t|v)a|jbro|jemu|jigs|kddi|keji|kgt( |\/)|klon|kpt |kwc\-|kyo(c|k)|le(no|xi)|lg( g|\/(k|l|u)|50|54|e\-|e\/|\-[a-w])|libw|lynx|m1\-w|m3ga|m50\/|ma(te|ui|xo)|mc(01|21|ca)|m\-cr|me(di|rc|ri)|mi(o8|oa|ts)|mmef|mo(01|02|bi|de|do|t(\-| |o|v)|zz)|mt(50|p1|v )|mwbp|mywa|n10[0-2]|n20[2-3]|n30(0|2)|n50(0|2|5)|n7(0(0|1)|10)|ne((c|m)\-|on|tf|wf|wg|wt)|nok(6|i)|nzph|o2im|op(ti|wv)|oran|owg1|p800|pan(a|d|t)|pdxg|pg(13|\-([1-8]|c))|phil|pire|pl(ay|uc)|pn\-2|po(ck|rt|se)|prox|psio|pt\-g|qa\-a|qc(07|12|21|32|60|\-[2-7]|i\-)|qtek|r380|r600|raks|rim9|ro(ve|zo)|s55\/|sa(ge|ma|mm|ms|ny|va)|sc(01|h\-|oo|p\-)|sdk\/|se(c(\-|0|1)|47|mc|nd|ri)|sgh\-|shar|sie(\-|m)|sk\-0|sl(45|id)|sm(al|ar|b3|it|t5)|so(ft|ny)|sp(01|h\-|v\-|v )|sy(01|mb)|t2(18|50)|t6(00|10|18)|ta(gt|lk)|tcl\-|tdg\-|tel(i|m)|tim\-|t\-mo|to(pl|sh)|ts(70|m\-|m3|m5)|tx\-9|up(\.b|g1|si)|utst|v400|v750|veri|vi(rg|te)|vk(40|5[0-3]|\-v)|vm40|voda|vulc|vx(52|53|60|61|70|80|81|83|85|98)|w3c(\-| )|webc|whit|wi(g |nc|nw)|wmlb|wonu|x700|xda(\-|2|g)|yas\-|your|zeto|zte\-'
    _short_matches = re.compile(_short_matches, re.IGNORECASE)

    if _long_matches.search(userAgent) != None:
        return True
    user_agent = userAgent[0:4]
    if _short_matches.search(user_agent) != None:
        return True
    return False


def getNonceStr(length = 16):
    #获取noncestr（随机字符串）
    return ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(15))

